how to filter spark dataframe using pyspark